home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 2 / Apprentice-Release2.iso / Tools / Languages / Python 1.1 / Demo / dns / dnslib.py < prev    next >
Encoding:
Text File  |  1994-10-08  |  14.9 KB  |  101 lines  |  [TEXT/R*ch]
  1. # Domain Name Server (DNS) interface
  2. #
  3. # See RFC 1035:
  4. # ------------------------------------------------------------------------
  5. # Network Working Group                                     P. Mockapetris
  6. # Request for Comments: 1035                                           ISI
  7. #                                                            November 1987
  8. # Obsoletes: RFCs 882, 883, 973
  9. # 
  10. #             DOMAIN NAMES - IMPLEMENTATION AND SPECIFICATION
  11. # ----------------------------------------------------------dnslib.py',
  12.         print '[-T] [-r] [-s server] [-t] [-u]',
  13.         print '[qtype [qname]]'
  14.         print '-T:        run testpacker() and exit'
  15.         print '-r:        recursion desired (default not)'
  16.         print '-s server: use server (default %s)' % server
  17.         print '-t:        use TCP protocol'
  18.         print '-u:        use UDP protocol (default)'
  19.         print 'qtype:     query type (default %s)' % \
  20.               dnstype.typestr(qtype)
  21.         print 'qname:     query name (default %s)' % qname
  22.         print 'Recognized qtype values:'
  23.         qtypes = dnstype.typemap.keys()
  24.         qtypes.sort()
  25.         n = 0
  26.         for qtype in qtypes:
  27.             n = n+1
  28.             if n >= 8: n = 1; print
  29.             print '%s = %d' % (dnstype.typemap[qtype], qtype),
  30.         print
  31.         sys.exit(2)
  32.     for o, a in opts:
  33.         if o == '-T': testpacker(); return
  34.         if o == '-t': protocol = 'tcp'
  35.         if o == '-u': protocol = 'udp'
  36.         if o == '-s': server = a
  37.         if o == '-r': rd = 1
  38.     if args[0:]:
  39.         try:
  40.             qtype = eval(string.upper(args[0]), dnstype.__dict__)
  41.         except (NameError, SyntaxError):
  42.             print 'bad query type:', `args[0]`
  43.             sys.exit(2)
  44.     if args[1:]:
  45.         qname = args[1]
  46.     if qtype == dnstype.AXFR:
  47.         print 'Query type AXFR, protocol forced to TCP'
  48.         protocol = 'tcp'
  49.     print 'QTYPE %d(%s)' % (qtype, dnstype.typestr(qtype))
  50.     m = Mpacker()
  51.     m.addHeader(0,
  52.           0, opcode, 0, 0, rd, 0, 0, 0,
  53.           1, 0, 0, 0)
  54.     m.addQuestion(qname, qtype, dnsclass.IN)
  55.     request = m.getbuf()
  56.     if protocol == 'udp':
  57.         s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
  58.         s.connect((server, port))
  59.         s.send(request)
  60.         reply = s.recv(1024)
  61.     else:
  62.         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  63.         s.connect((server, port))
  64.         s.send(pack16bit(len(request)) + request)
  65.         s.shutdown(1)
  66.         f = s.makefile('r')
  67.         header = f.read(2)
  68.         if len(header) < 2:
  69.             print '*** EOF ***'
  70.             return
  71.         count = unpack16bit(header)
  72.         reply = f.read(count)
  73.         if len(reply) != count:
  74.             print '*** Incomplete reply ***'
  75.             return
  76.     u = Munpacker(reply)
  77.     dumpM(u)
  78.     if protocol == 'tcp' and qtype == dnstype.AXFR:
  79.         while 1:
  80.             header = f.read(2)
  81.             if len(header) < 2:
  82.                 print '========== EOF =========='
  83.                 break
  84.             count = unpack16bit(header)
  85.             if not count:
  86.                 print '========== ZERO COUNT =========='
  87.                 break
  88.             print '========== NEXT =========='
  89.             reply = f.read(count)
  90.             if len(reply) != count:
  91.                 print '*** Incomplete reply ***'
  92.                 break
  93.             u = Munpacker(reply)
  94.             dumpM(u)
  95.  
  96.  
  97. # Run test program when called as a script
  98.  
  99. if __name__ == '__main__':
  100.     test()
  101.